import pandas as pd

data=pd.read_csv('../data_report/report-ml.csv')

data_len = len(data)
# 训练集数量
lens = 5000

if data_len > lens:      
    # 拆分出训练集
    data[data_len-lens:].to_csv('../data_train/report_train.csv',index=False)
    # 拆分出测试集
    data[:data_len-lens].to_csv('../data_test/report_test.csv',index=False)
